@echo off
set MID_DATA_DIR=./data/mid_data
set RAW_DATA_DIR=./data/raw_data
set OUTPUT_DIR=./out

set GPU_IDS=0
set BERT_TYPE=roberta_wwm
set BERT_DIR=../bert/torch_%BERT_TYPE%

set MODE=train
set TASK_TYPE=crf

python main.py ^
--gpu_ids=%GPU_IDS% ^
--output_dir=%OUTPUT_DIR% ^
--mid_data_dir=%MID_DATA_DIR% ^
--mode=%MODE% ^
--task_type=%TASK_TYPE% ^
--raw_data_dir=%RAW_DATA_DIR% ^
--bert_dir=%BERT_DIR% ^
--bert_type=%BERT_TYPE% ^
--train_epochs=10 ^
--swa_start=5 ^
--attack_train= ^
--train_batch_size=8 ^
--dropout_prob=0.1 ^
--max_seq_len=256 ^
--lr=2e-5 ^
--other_lr=2e-3 ^
--seed=123 ^
--weight_decay=0.01 ^
--loss_type=ls_ce ^
--eval_model